package com.heima.media.service.impl;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.heima.common.dto.PageResponseResult;
import com.heima.common.dto.ResponseResult;
import com.heima.common.dto.User;
import com.heima.common.enums.AppHttpCodeEnum;
import com.heima.common.util.MediaThreadLocalUtil;
import com.heima.media.dto.ContentDto;
import com.heima.media.dto.ImageDto;
import com.heima.media.dto.WmNewsDto;
import com.heima.media.dto.WmNewsPageDto;
import com.heima.media.entity.WmMaterial;
import com.heima.media.entity.WmNews;
import com.heima.media.entity.WmNewsMaterial;
import com.heima.media.mapper.WmNewsMapper;
import com.heima.media.service.IWmNewsMaterialService;
import com.heima.media.service.IWmNewsService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * <p>
 * 自媒体图文内容信息表 服务实现类
 * </p>
 *
 * @author mcm
 * @since 2022-08-09
 */
@Service
public class WmNewsServiceImpl extends ServiceImpl<WmNewsMapper, WmNews> implements IWmNewsService {

    @Autowired
    private IWmNewsMaterialService newsMaterialService;

    //自媒体文章列表查询
    @Override
    public ResponseResult listByCondition(WmNewsPageDto dto) {
        //构造查询条件
        LambdaQueryWrapper<WmNews> query = new LambdaQueryWrapper<>();
        //根据文章状态查询
        if (dto.getStatus() != null) {
            query.eq(WmNews::getStatus, dto.getStatus());
        }
        //根据关键字模糊查询文章标题
        if (!StringUtils.isEmpty(dto.getKeyword())) {
            query.like(WmNews::getTitle, dto.getKeyword());
        }
        //根据频道列表查询
        if (dto.getChannelId() != null) {
            query.eq(WmNews::getChannelId, dto.getChannelId());
        }
        // 发布日期 开始日期  -- 结束日期
        if (dto.getBeginPubDate() != null) {
            query.gt(WmNews::getPublishTime, dto.getBeginPubDate());
        }
        if (dto.getEndPubDate() != null) {
            query.lt(WmNews::getPublishTime, dto.getEndPubDate());
        }
        // 当前用户
        User user = MediaThreadLocalUtil.get();
        if (user == null) {
            return ResponseResult.errorResult(AppHttpCodeEnum.NEED_LOGIN);
        }
        query.eq(WmNews::getUserId, user.getUserId());
        // 按照文章发布时间倒序排列
        query.orderByDesc(WmNews::getPublishTime);
        // 构建分页条件
        IPage<WmNews> page = new Page<>(dto.getPage(), dto.getSize());
        // 分页查询文章列表
        IPage<WmNews> iPage = this.page(page, query);
        // 构建通用的分页响应对象
        return new PageResponseResult(
                dto.getPage(),
                dto.getSize(),
                iPage.getTotal(),
                iPage.getRecords()
        );
    }

    @Override
    public ResponseResult submit(WmNewsDto dto) {
        //判断当前登录用户
        User user = MediaThreadLocalUtil.get();
        if (user == null) {
            return ResponseResult.errorResult(AppHttpCodeEnum.NEED_LOGIN);
        }
        //判断是否更新
        //如果是更新文章  ---dto中包含文章id   删除已关联素材关系
        if (dto.getId() == null) {
            deleteRelation(dto.getId());
        }

        //新增
        //构建 WmNews 对象，从dto中复制
        WmNews news = new WmNews();
        BeanUtils.copyProperties(dto, news);
        news.setUserId(user.getUserId());
        news.setCreatedTime(new Date());
        news.setSubmitedTime(new Date());
        //保存封面图片，将前端传递的集合转换成hson进行存储
        news.setImages(JSON.toJSONString(dto.getImages()));

        //如果前端传递的是自动  -1  需要从内容中提取图片作为封面
        List<ImageDto> contentImages = getImagesFromContent(dto.getContent());
        if (dto.getType() == -1) {
            if (contentImages.size() <= 0) {
                //如果内容中没有图片，修改类型为  0
                news.setType(0);
            } else if (contentImages.size() < 3) {
                //如果上传的图片数量<3，修改类型为单图  1  ，取一张图片作为封面图片
                news.setType(1);
                news.setImages(JSON.toJSONString(contentImages.subList(0, 1)));
            } else {
                //如果内容中图片数量>=3，修改类型为多图  3 ，取前三张作为封面图
                news.setType(3);
                List<ImageDto> images = contentImages.subList(0, 3);
                news.setImages(JSON.toJSONString(images));
            }
        }

        //保存文章
        //判断有没有id--->更新，没有id  新增
        this.saveOrUpdate(news);

        //如果是提交审核，需要保存文章和素材的引用关系 包含内容和素材的关系
        if (dto.getStatus() == 1) {
            //保存封面图片对应关系
            List<ImageDto> coverImages = JSON.parseArray(news.getImages(), ImageDto.class);
            saveRelation(news, contentImages, 0);
        }

        return ResponseResult.okResult();
    }

    /**
     * 根据文章id删除关联关系
     *
     * @param newsId
     */
    private void deleteRelation(Integer newsId) {
        LambdaQueryWrapper<WmNewsMaterial> query = new LambdaQueryWrapper<>();
        query.eq(WmNewsMaterial::getNewsId, newsId);
        newsMaterialService.remove(query);
    }

    /**
     * 从内容中提取图片
     *
     * @param content
     * @return
     */
    private List<ImageDto> getImagesFromContent(String content) {
        ArrayList<ImageDto> list = new ArrayList<>();
        //将内容转换成集合
        List<ContentDto> contentDtos = JSON.parseArray(content, ContentDto.class);
        //遍历集合，判断ytpe==image，获取value
        for (ContentDto contentDto : contentDtos) {
            if (contentDto.getType().equals("image")) {
                ImageDto imageDto = new ImageDto();
                imageDto.setId(contentDto.getId());
                imageDto.setUrl(contentDto.getValue());
                list.add(imageDto);
            }
        }
        return list;
    }

    /**
     * 保存图片对应关系
     *
     * @param news
     * @param images
     * @param type   1: 封面  0: 内容
     */
    private void saveRelation(WmNews news, List<ImageDto> images, int type) {
        Integer ord = 0;
        //构建图片对应关系
        //遍历集合
        for (ImageDto image : images) {
            WmNewsMaterial newsMaterial = new WmNewsMaterial();
            newsMaterial.setMaterialId(image.getId());
            newsMaterial.setNewsId(news.getId());
            newsMaterial.setType(type);
            newsMaterial.setOrd(ord);
            //保存
            newsMaterialService.save(newsMaterial);

            ord++;
        }
    }
}
